import os
import re


class TemplateDocUtil:
    def __init__(self, filename):
        self.filename = filename

    def parseMemo(self, str):
        _pattern = re.compile(r'@Memo\(.*\)')  # 匹配至少一个数字
        _result = _pattern.findall(str)
        arr = []
        for option in _result:
            arr.append(option[6:-1])
        return arr

    def getFuncDefine(self):
        with open(self.filename,"r", encoding="utf8") as f:
            # 读取文件中全部内容
            lines = f.readlines()
            list = []
            # 读取文件中的每一行
            for line in lines:
                # 对每一行的内容进行一些字符串操作（包括去除字符串左右两边换行符和'w'，和空格）。
                if line.find("@Memo(") >=0:
                    list.append(self.parseMemo(line.strip().replace("\n","")))
            # 输出list观察是否符合预期
            return (list)